package com.qk.management.web.filter;

import cn.hutool.core.util.ObjectUtil;
import com.qk.common.util.JwtUtil;
import io.jsonwebtoken.Claims;
import jakarta.servlet.*;
import jakarta.servlet.annotation.WebFilter;
import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;
import lombok.extern.slf4j.Slf4j;

import java.io.IOException;

@Slf4j
//@WebFilter("/*")
public class LoginCheckFilter implements Filter {

    private static final String LOGIN_PATH = "/login";
    private static final String TOKEN_HEADER_NAME = "token";

    @Override
    public void doFilter(ServletRequest req, ServletResponse res, FilterChain chain) throws IOException, ServletException {
        HttpServletRequest request =  (HttpServletRequest) req;
        HttpServletResponse response = (HttpServletResponse) res;

        String path = request.getRequestURI();

        boolean isLogin = ObjectUtil.equals(LOGIN_PATH, path);
        if (isLogin){
            chain.doFilter(request, response);
            return;
        }

        String token = request.getHeader(TOKEN_HEADER_NAME);

        if (ObjectUtil.isEmpty( token)){
            response.setStatus(401);
            return;
        }

        try{
            Claims claims =  JwtUtil.parseToken(token);
            String username = claims.get("username", String.class);
            Object id = claims.get("id");
            chain.doFilter(request, response);
            return;
        }catch (Exception e){
            response.setStatus(401);
            return;
        }
    }
}
